#!/bin/bash

source /root/anaconda3/etc/profile.d/conda.sh

conda activate omnigen


accelerate launch --num_processes=1 /home/bsliu/gitprojects/OmniGen/train.py \
    --model_name_or_path /home/bsliu/gitprojects/OmniGen/checkpoint/omnigen-v1 \
    --batch_size_per_device 5 \
    --condition_dropout_prob 0.01 \
    --lr 4e-5 \
    --json_file /share/project/dataset_raw/imagenet/train/filelist.txt \
    --image_path /home/bsliu/gitprojects/OmniGen/toy_data/images \
    --max_input_length_limit 18000 \
    --keep_raw_resolution \
    --max_image_size 1024 \
    --gradient_accumulation_steps 1 \
    --ckpt_every 1000 \
    --epochs 10 \
    --log_every 10 \
    --results_dir /home/bsliu/gitprojects/OmniGen/results/imagenet
